import torch.nn as nn
import torch

from backbone import d
from backbone import darknet53

if __name__ == '__main__':
    a = d.YOLOModel(3 * (5 + 1))
    b = darknet53.YoloBody([3, 3, 3], 1)
    print(len(list(a.state_dict())))
    print(len(list(b.state_dict())))
